import os
import random

# 设置文件目录
input_dir = r'C:\Users\ASUS\Desktop\2024年新生赛出题\2024年新生赛出题(非大一组)\H题\H题_input'
os.makedirs(input_dir, exist_ok=True)

# 生成随机测试数据
def generate_test_case():
    # 随机生成 m 和 n
    m = random.randint(1, 300)
    n = random.randint(1, 300)

    # 生成 map 数据，随机填充 '0' 和 '1'
    map_data = []
    for _ in range(n):
        row = ''.join(random.choice('01') for _ in range(m))
        map_data.append(row)

    return m, n, map_data

# 生成 15 组测试数据
for idx in range(1, 16):
    m, n, map_data = generate_test_case()
    
    # 保存到 .in 文件
    input_path = os.path.join(input_dir, f'{idx}.in')
    with open(input_path, 'w') as f:
        f.write(f"{m} {n}\n")
        f.write("\n".join(map_data) + "\n")

print("15组测试数据已生成并保存在指定目录中。")
